from __future__ import absolute_import
from __future__ import print_function

import os
import json
from keras.optimizers import Adam, SGD
from Tested_models.nn_regression.custom import weighted_mean_squared_error,auc_obj,mean_squared_error,binary_crossentropy_with_ranking

from meta import TrainingMetadata
from experiment import TRAINING_JSON, META_DIR


def get_training_metadata():
    training_metadata = TrainingMetadata()
    training_metadata.optimizer = SGD(momentum=0.9, decay=0.9, lr=0.01)
    training_metadata.loss = binary_crossentropy_with_ranking
    training_metadata.batch_size = 48
    training_metadata.nb_epoch = 5
    training_metadata.nb_iter = 1
    training_metadata.shuffle = False
    training_metadata.mode = 'classify'


    return training_metadata
